package com.hunter;

import org.w3c.dom.traversal.NodeIterator;

import java.util.Iterator;
import java.util.function.Consumer;

// 单项链表
public class DanXiangLianBiao implements Iterable<Integer> { // 整体
    private Node head = null; // 头指针

    @Override
    public Iterator<Integer> iterator() {
        // 匿名内部类 -> 带名字内部类
        return new IntegerIterator();
    }


    /*
    节点类
     */
    private static class Node {
        int value; // 值
        Node next; // 下一个节点指针

        public Node(int value, Node next) {
            this.value = value;
            this.next = next;
        }
    }

    /*
    向链表头部添加
    value - 待加值
     */
    public void addFirst(int value) {
        // 1.链表为空
//        head = new Node(value, null);  头指针不存储数据,只指向下一个元素
        // 2.链表非空

        head = new Node(value, head);
    }

    /*
    遍历链表（一）
     */
    public void loop1(Consumer<Integer> consumer) {
        Node p = head;
        while (p != null) {
            consumer.accept(p.value);
            p = p.next;
        }
    }

    /*
    遍历链表（二）
     */
    public void loop2(Consumer<Integer> consumer) {
        for (Node p = head; p != null; p = p.next) {
            consumer.accept(p.value);
        }
    }

    private class IntegerIterator implements Iterator<Integer> {
        Node p = head;

        @Override
        public boolean hasNext() { // 是否有下一个元素
            return p != null;
        }

        @Override
        public Integer next() { // 返回当前值，并指向下一个元素
            int v = p.value;
            p = p.next;
            return v;
        }
    }

    private Node findLast() {
        // 空链表
        if (head == null) {
            return null;
        }
        Node p;
        // 当元素下一个节点值空时，不断取下一个节点的值
        // 最后取最后一个节点的值
        for (p = head; p.next != null; p = p.next) {

        }
        return p;
    }

    public void addLast(int value) {
        Node last = findLast();
        if (last == null) {
            addFirst(value);
            return;
        }
        last.next = new Node(value, null);
    }

    //    public void test() {
//        int i = 0;
//        for (Node p = head; p != null; p = p.next, i++) {
//            System.out.println(p.value + "的索引是:" + i);
//        }
//    }

    // 查找节点
    private Node findNode(int index) {
        int i = 0;
        for (Node p = head; p != null; p = p.next, i++) {
            if (i == index) {
                return p;
            }
        }
        return null; // 没找到
    }

    /*
    根据索引查找节点的值
     */
    public int get(int index) {
        Node node = findNode(index);
        if (node == null) {
            throw illegalIndex(index);
        }
        return node.value;
    }

    /*
    向索引位置插入
     */
    public void insert(int index, int value) {
        Node prev = findNode(index - 1); // 找上一个节点
        if (prev == null) { // 找不到
            throw illegalIndex(index);
        }
        prev.next = new Node(value, prev.next);
    }

    /*
    不合法 index 异常
     */
    private static IllegalArgumentException illegalIndex(int index) {
        return new IllegalArgumentException(
                String.format("index [%d] 不合法%n", index)
        );
    }

    /*
    删除第一个节点
    如果不存在，则抛出index非法异常
     */
    public void removeFirst() {
        if (head == null) {
            throw illegalIndex(0);
        }
        head = head.next;


    }

    /*
    按照index删除节点
     */
    public void remove(int index) {

        // 如果index为0，则删除第一个节点
        if (index == 0) {
            removeFirst();
            return;
        }
        Node prev = findNode(index - 1); // 上一个节点

        // 如果上一个节点为null，抛出index异常
        if (prev == null) {
            throw illegalIndex(index);
        }
        Node remove = prev.next; // 被删除的节点

        // 如果要删除的节点为null，抛出index异常
        if (remove == null) {
            throw illegalIndex(index);
        }
        prev.next = remove.next; // 上一个节点指向节点 = 被删除的节点所指向节点
    }


}
